import unittest

def add(a, b):
    return a + b

class TestAdd(unittest.TestCase):
    def test_add01(self):
        result = add(1, 2)
        # 断言：期望结果与实际结果是否一致
        assert result == 3
    def test_add02(self):
        result = add(3, 4)
        # 断言结果是7
        self.assertEqual(result, 7)
    def test_add03(self):
        result = add(5, 6)
        is_ok = result == 11
        # 断言结果是否为True
        self.assertTrue(is_ok)
    def test_add04(self):
        # is_ok = True
        # self.assertFalse(is_ok)
        msg = '登录成功！欢迎回来：xxx'
        # 断言字符串是否出在指定的内容中
        # self.assertIn('登录成功', msg)
        # 断言字符串是否不在指定的内容中
        # self.assertNotIn('abc', 'hello world')
        n = None
        # 断言一个变量是否为None
        # self.assertIsNone(n)
        # 断言一个变量是否不为None
        self.assertIsNotNone(n)

if __name__ == '__main__':
    unittest.main()
